package com.dingwen.treasure.base.util;

import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.CollectionUtils;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * 扩展Spring的BeanUtils工具类
 *
 * @author dingwen
 * @date 2023/2/18 15:50
 */
@Slf4j
public class BeanUtil extends org.springframework.beans.BeanUtils {
    /**
     * 匹配getter方法的正则表达式
     */
    private static final Pattern GET_PATTERN = Pattern.compile("get(\\p{javaUpperCase}\\w*)");

    /**
     * 匹配setter方法的正则表达式
     */
    private static final Pattern SET_PATTERN = Pattern.compile("set(\\p{javaUpperCase}\\w*)");

    /**
     * Bean属性复制工具方法。
     *
     * @param target 目标对象
     * @param source 源对象
     */
    public static void copyProperties(Object source, Object target) {
        try {
            org.springframework.beans.BeanUtils.copyProperties(source, target);
        } catch (Exception e) {
            log.error("拷贝对象数据异常：{},{},错误消息{}",
                    source.getClass().getName(),
                    target.getClass().getName(),
                    e.getMessage());
        }
    }

    /**
     * 获取对象的setter方法。
     *
     * @param obj 对象
     * @return 对象的setter方法列表
     */
    public static List<Method> getSetterMethods(Object obj) {
        // setter方法列表
        List<Method> setterMethods = new ArrayList<>();

        // 获取所有方法
        Method[] methods = obj.getClass().getMethods();

        // 查找setter方法

        for (Method method : methods) {
            Matcher m = SET_PATTERN.matcher(method.getName());
            if (m.matches() && (method.getParameterTypes().length == 1)) {
                setterMethods.add(method);
            }
        }
        // 返回setter方法列表
        return setterMethods;
    }

    /**
     * 获取对象的getter方法。
     *
     * @param obj 对象
     * @return 对象的getter方法列表
     */

    public static List<Method> getGetterMethods(Object obj) {
        // getter方法列表
        List<Method> getterMethods = new ArrayList<>();
        // 获取所有方法
        Method[] methods = obj.getClass().getMethods();
        // 查找getter方法
        for (Method method : methods) {
            Matcher m = GET_PATTERN.matcher(method.getName());
            if (m.matches() && (method.getParameterTypes().length == 0)) {
                getterMethods.add(method);
            }
        }
        // 返回getter方法列表
        return getterMethods;
    }

    /**
     * 根据class复制属性
     *
     * @param source 源
     * @param clazz  clazz
     * @return 根据给定的class类生成的对象
     */
    @SneakyThrows(Throwable.class)
    public static <R, T> T copyProperties(R source, Class<T> clazz) {
        T target = clazz.newInstance();
        copyProperties(source, clazz.newInstance());
        return target;
    }

    /**
     * 根据class复制出ArrayList
     *
     * @param source 源
     * @param clazz  clazz
     * @return {@link List}<{@link T}>根据给定的class类生成的数组
     */
    public static <R, T> List<T> copyProperties(List<R> source, Class<T> clazz) {
        List<T> list = new ArrayList<T>(source.size());
        if (!CollectionUtils.isEmpty(source)) {
            source.forEach(r -> list.add(copyProperties(r, clazz)));
        }
        return list;
    }
}
